package com.xy.work.template.base.utils;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.xy.work.template.base.domain.LoginUser;
import com.xy.work.template.common.enums.CommonCodeEnum;
import com.xy.work.template.common.exception.BaseCustomException;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import java.util.Date;
/**
 * @author niuniu
 * @Date: 2021/8/15 07:38
 * @Description:
 */
@Slf4j
public class JwtUtil {
    /**
     * token 过期时间
     */
    private static  long EXPIRE_DAY;

    /**
     * 加密的秘钥
     */
    private static String SECRET;

    /**
     * 令牌前缀
     */
    private static String TOKEN_PREFIX;

    /**
     * 项目名称
     */
    private static String SUBJECT;
    /**
     * 用户头像键名称
     */
    private static final String LOGIN_PARAMETER_HEAD_IMG_KEY ="head_img";
    /**
     * 用户昵称键名称
     */
    private static final String LOGIN_PARAMETER_NAME_KEY ="name";
    /**
     * 用户ID键名称
     */
    private static final String LOGIN_PARAMETER_ID_KEY ="id";
    /**
     * 扩展参数键名称
     */
    private static final String LOGIN_PARAMETER_EXTEND_KEY ="EXTEND";

    /**
     * 初始化
     * @param expireDay token 过期时间 单位天
     * @param tokenPrefix 令牌前缀
     * @param subject 项目名称
     * @param secret 加密的秘钥
     */
    public static void init(Long expireDay,String tokenPrefix,String subject,String secret){
        EXPIRE_DAY=expireDay;
        TOKEN_PREFIX=tokenPrefix;
        SUBJECT=subject;
        SECRET=secret;
        check();

    }

    /**
     * 公共检测
     */
    private static void check(){
        if(EXPIRE_DAY <= 0){
            throw new BaseCustomException(CommonCodeEnum.JWT_ERROR);
        }
        if(StrUtil.isEmpty(TOKEN_PREFIX)){
            throw new BaseCustomException(CommonCodeEnum.JWT_ERROR);
        }
        if(StrUtil.isEmpty(SUBJECT)){
            throw new BaseCustomException(CommonCodeEnum.JWT_ERROR);
        }
        if(StrUtil.isEmpty(SECRET)){
            throw new BaseCustomException(CommonCodeEnum.JWT_ERROR);
        }
    }
    /**
     * 根据用户信息，生成令牌
     * @param loginUser
     * @return
     */
    public static String geneJsonWebToken(LoginUser loginUser) {
        if (loginUser == null) {
            throw new NullPointerException("loginUser对象为空");
        }
        check();
        String token = Jwts.builder().setSubject(SUBJECT)
                //payload
                .claim(LOGIN_PARAMETER_HEAD_IMG_KEY, loginUser.getHeadImg())
                .claim(LOGIN_PARAMETER_ID_KEY, loginUser.getId())
                .claim(LOGIN_PARAMETER_NAME_KEY, loginUser.getName())
                .claim(LOGIN_PARAMETER_EXTEND_KEY, loginUser.getExtend())
                .setIssuedAt(new Date())
                .setExpiration(getExpiration())
                .signWith(SignatureAlgorithm.HS256, SECRET).compact();

        token = TOKEN_PREFIX + token;
        return token;
    }

    /**
     * 得到过期有效时间，单位毫秒
     * @return
     */
    public static Date getExpiration(){
        return new Date(System.currentTimeMillis() + (1000 * 60 * 60 * 24 * EXPIRE_DAY));
    }

    /**
     * 校验token的方法
     *
     * @param token
     * @return
     */
    private static Claims checkJwtClaims(String token) {
        try {
            check();
            return Jwts.parser()
                    .setSigningKey(SECRET)
                    .parseClaimsJws(token.replace(TOKEN_PREFIX, "")).getBody();
        } catch (Exception e) {
            log.error("jwt token解密失败",e);
            return null;
        }

    }

    /**
     * 校验token的方法
     *
     * @param token
     * @return
     */
    public static LoginUser checkJwt(String token) {
        try {
            LoginUser loginUser = new LoginUser();
            Claims claims = checkJwtClaims(token);
            log.info("解析登陆授权信息:{}",claims);
            if(ObjectUtil.isNotNull(claims)){
                loginUser.setHeadImg(String.valueOf(claims.get(LOGIN_PARAMETER_HEAD_IMG_KEY)));
                loginUser.setExtend(claims.get(LOGIN_PARAMETER_EXTEND_KEY));
                loginUser.setId(Long.parseLong(String.valueOf(claims.get(LOGIN_PARAMETER_ID_KEY))));
                loginUser.setName(String.valueOf(claims.get(LOGIN_PARAMETER_NAME_KEY)));
                return loginUser;
            }else{
                throw new BaseCustomException(CommonCodeEnum.LOGIN_EXPIRE);
            }
        } catch (Exception e) {
            throw e;
        }
    }
}
